python list comprehension multiple if statements